from zhangUtils.util import *
import argparse

parser = argparse.ArgumentParser(description="Deep Multi-Patch Hierarchical Network")
parser.add_argument("-e", "--epochs", type=int, default=10000)
parser.add_argument("-se", "--start_epoch", type=int, default=200)
parser.add_argument("-b", "--batchsize", type=int, default=8)
parser.add_argument("-c", "--cropsize", type=int, default=128)
parser.add_argument("-l", "--learning_rate", type=float, default=0.0001)
parser.add_argument("--data_path", type=str, default='/root/project/dataset/data', help="the path of NH-HAZE dataset")
parser.add_argument("--checkpoint_path", type=str, default='/root/project/code/hrs-mdsf/checkpoint/checkpoint_2114.pth', help="path to save checkpoints")
args = parser.parse_args()

if __name__ == '__main__':
    ssim_val, psnr_val = evaluation(args.save_dir, args.clean_dir)

    print('ssim: ', ssim_val)
    print('psnr: ', psnr_val)
